package com.iotdb.zjc.demo.util;

import org.apache.kafka.clients.admin.AdminClient;
import org.apache.kafka.clients.admin.AdminClientConfig;
import org.apache.kafka.clients.admin.NewTopic;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.kafka.core.KafkaAdmin;

import java.util.HashMap;
import java.util.Map;

@Configuration
public class KafkaInitialConfiguration {

    @Value("${spring.kafka.bootstrap-servers}")
    private String bootstrapServers;
    /**
     * 创建TopicName为topic.quick.initial的Topic并设置分区数为8以及副本数为1
     * 通过bean创建(bean的名字为initialTopic)
     * @return
     */
//    @Bean
//    public NewTopic initialTopic() {
//        return new NewTopic("TopicC", 8, (short) 1);
//    }
//
//    /**
//     * 此种@Bean的方式，如果topic的名字相同，那么会覆盖以前的那个
//     * 修改后|分区数量会变成11个 注意分区数量只能增加不能减少
//     * @return
//     */
//    @Bean
//    public NewTopic initialTopic2() {
//        return new NewTopic("TopicC", 11, (short) 1);
//    }

    /**
     * 创建一个kafka管理类，相当于rabbitMQ的管理类rabbitAdmin,没有此bean无法自定义的使用adminClient创建topic
     * @return
     */
    @Bean
    public KafkaAdmin kafkaAdmin() {
        Map<String, Object> props = new HashMap<>();
        //配置Kafka实例的连接地址 kafka的地址，不是zookeeper
        props.put(AdminClientConfig.BOOTSTRAP_SERVERS_CONFIG, bootstrapServers);
        KafkaAdmin admin = new KafkaAdmin(props);
        return admin;
    }

    /**
     * kafka客户端，在spring中创建这个bean之后可以注入并且创建topic
     * @return
     */
    @Bean
    public AdminClient adminClient() {
        return AdminClient.create(kafkaAdmin().getConfig());
    }


}